/*
 * Copyright © 2018 www.noark.xyz All Rights Reserved.
 *
 * 感谢您选择Noark框架，希望我们的努力能为您提供一个简单、易用、稳定的服务器端框架 ！
 * 除非符合Noark许可协议，否则不得使用该文件，您可以下载许可协议文件：
 *
 *        http://www.noark.xyz/LICENSE
 *
 * 1.未经许可，任何公司及个人不得以任何方式或理由对本框架进行修改、使用和传播;
 * 2.禁止在本项目或任何子项目的基础上发展任何派生版本、修改版本或第三方版本;
 * 3.无论你对源代码做出任何修改和改进，版权都归Noark研发团队所有，我们保留所有权利;
 * 4.凡侵犯Noark版权等知识产权的，必依法追究其法律责任，特此郑重法律声明！
 */
package xyz.noark.orm.accessor.adaptor;

import org.bson.Document;
import xyz.noark.orm.FieldMapping;
import xyz.noark.orm.accessor.sql.PreparedStatementProxy;

import java.io.IOException;
import java.sql.ResultSet;
import java.util.concurrent.atomic.LongAdder;

/**
 * LongAdder类型属性
 *
 * @author 小流氓[176543888@qq.com]
 * @since 3.2.6
 */
class LongAdderAdaptor extends AbstractValueAdaptor<LongAdder> {

    @Override
    protected void toPreparedStatement(FieldMapping fm, PreparedStatementProxy pstmt, LongAdder value, int parameterIndex) throws Exception {
        pstmt.setLong(parameterIndex, value == null ? 0L : value.longValue());
    }

    @Override
    protected Object toParameter(FieldMapping fm, ResultSet rs) throws Exception {
        final LongAdder adder = new LongAdder();
        adder.add(rs.getLong(fm.getColumnName()));
        return adder;
    }

    @Override
    protected Object readGeneratedValue(FieldMapping fm, ResultSet rs) throws Exception {
        final LongAdder adder = new LongAdder();
        adder.add(rs.getLong(1));
        return adder;
    }

    @Override
    protected void doPersistFieldToStorage(Document document, String columnName, FieldMapping fm, LongAdder columnValue) throws Exception {
        document.append(columnName, this.toLongValue(columnValue));
    }

    private long toLongValue(LongAdder columnValue) {
        return columnValue == null ? 0L : columnValue.longValue();
    }

    @Override
    protected Object doRetrieveFieldFromStorage(Document document, FieldMapping fm, String columnName) throws IOException {
        final LongAdder adder = new LongAdder();
        adder.add(document.getLong(columnName));
        return adder;
    }
}